from algorithm import *

def test1():
    myDat, labels = createDataSet()
    ent = calcShannonEnt(myDat)
    print(ent)

def test2():
    myDat, labels = createDataSet()
    dat1 = splitDataSet(myDat, 0, 1)
    dat2 = splitDataSet(myDat, 0, 0)
    print(dat1)
    print(dat2)

def test3():
    myDat, labels = createDataSet()
    print(myDat, labels)
    best = chooseBestFeatureToSplit(myDat)
    print("BestFeature:" + str(best))

#test1()
#test2()
test3()